from mindspeed.te.pytorch.module import MindSpeedTELayernorm